package com.sen.service.impl;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.sen.dao.DrugtypeDao;
import com.sen.entity.DruginfoEntity;
import com.sen.entity.DrugtypeEntity;
import com.sen.entity.MenuEntity;
import com.sen.service.DrugtypeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;

@Service("drugtypeService")
public class DrugtypeServiceImpl extends ServiceImpl<DrugtypeDao, DrugtypeEntity> implements DrugtypeService {

    @Autowired
    DrugtypeDao drugtypeDao;

    @Override
    public List<DrugtypeEntity> queryAllDrugType(Map<String, String> drugtype) {
        List<DrugtypeEntity> druglist = drugtypeDao.selectList(new QueryWrapper<DrugtypeEntity>().eq("pid", 0));
        return twoDrugtypeEntityList(druglist);
    }
    public List<DrugtypeEntity> twoDrugtypeEntityList(List<DrugtypeEntity> drugtypes){
        for (DrugtypeEntity drugtype : drugtypes) {

            List<DrugtypeEntity> drugtypeEntities = drugtypeDao.selectList(new QueryWrapper<DrugtypeEntity>().eq("pid", drugtype.getTid()));
            drugtype.setChildren(drugtypeEntities);
            if(drugtype.getChildren()!= null && drugtype.getChildren().size() > 0){
                twoDrugtypeEntityList(drugtypeEntities);
            }
        }
        return drugtypes;
    }

}
